package com.example.demo.leetcode;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 链接：<a href="https://leetcode.cn/problems/merge-intervals/?envType=study-plan-v2&envId=top-interview-150">https://leetcode.cn/problems/merge-intervals/?envType=study-plan-v2&envId=top-interview-150</a>
 * <p>
 * 以数组 intervals 表示若干个区间的集合，其中单个区间为 intervals[i] = [start, end] 。请你合并所有重叠的区间，并返回 一个不重叠的区间数组，该数组需恰好覆盖输入中的所有区间 。
 * <p>
 * 示例 1：
 * <p>
 * 输入：intervals = [[1,3],[2,6],[8,10],[15,18]] <br>
 * 输出：[[1,6],[8,10],[15,18]] <br>
 * 解释：区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6] <br>
 * <p>
 * <p>
 * 示例 2：
 * <p>
 * 输入：intervals = [[1,4],[4,5]] <br>
 * 输出：[[1,5]] <br>
 * 解释：区间 [1,4] 和 [4,5] 可被视为重叠区间。 <br>
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= intervals.length <= 10^4 <br>
 * intervals[i].length == 2 <br>
 * 0 <= start <= end <= 10^4 <br>
 */
public class _56_合并区间 {

    public static void main(String[] args) {
        int[][] intervals = {{1, 3}, {2, 6}, {8, 10}, {15, 18}};
        int[][] merge = new Solution().merge(intervals);
        System.out.println(Arrays.deepToString(merge));
    }

    public static class Solution {
        /**
         * 解析思路：<br>
         * 先将二维数组排序，按照数组中首个数字的大小，从小到大进行排序。<br>
         * 排序完成之后，连续的区间肯定是相邻的。<br>
         * <p>
         * 时间复杂度：O(nlogn+n)<br>
         * 空间复杂度:O(n*2);
         *
         * @author WangYX
         * @date 2024/01/24 16:57
         * @version 1.0.0
         */
        public int[][] merge(int[][] intervals) {
            List<int[]> list = new ArrayList<>();
            //排序
            quickSort(intervals, 0, intervals.length - 1);
            int[] temp = intervals[0];
            for (int i = 1; i < intervals.length; i++) {
                //判断元素temp数组的第二个值是否大于当前遍历元素的第一个值
                //如果大于则说明temp数组的值不用改变。否则，说明两个区间不能连续。
                if (temp[1] >= intervals[i][0]) {
                    //在大于当前遍历元素的第一个值的基础上，再次判断是否小于第二个值。
                    //小于就修改tem的第二个值。否则不动。
                    if (temp[1] < intervals[i][1]) {
                        temp[1] = intervals[i][1];
                    }
                } else {
                    //区间不连续，则把temp存入list
                    list.add(temp);
                    //再把temp指向当前元素
                    temp = intervals[i];
                }
            }
            //最后temp写入list中
            list.add(temp);
            return list.toArray(new int[list.size()][]);
        }

        /**
         * 按照二维数组的第一个元素进行快速排序
         */
        private void quickSort(int[][] intervals, int low, int hight) {
            if (low < hight) {
                int partition = partition(intervals, low, hight);
                //基准元素左边递归
                quickSort(intervals, low, partition - 1);
                //基准元素右边递归
                quickSort(intervals, partition + 1, hight);
            }
        }

        /**
         * 把数组分块
         */
        private int partition(int[][] intervals, int low, int hight) {
            int privot = intervals[low][0];
            int[] interval = intervals[low];
            while (low < hight) {
                while (low < hight && intervals[hight][0] >= privot) {
                    hight--;
                }
                intervals[low] = intervals[hight];
                while (low < hight && intervals[low][0] <= privot) {
                    low++;
                }
                intervals[hight] = intervals[low];
            }
            intervals[low] = interval;
            return low;
        }
    }
}
